Ubuntu20.04 賷亘賳賷 pytorch1.8 亘賷卅丞 丕賱鬲毓賱賲 丕賱毓賲賷賯 |
您所在的位置：网站首页 › ubuntu2004 vscode › Ubuntu20.04 賷亘賳賷 pytorch1.8 亘賷卅丞 丕賱鬲毓賱賲 丕賱毓賲賷賯 |
|
1. 賳爻禺丞 丕賱亘賷卅丞
Ubuntu 20.04
Python 3.8 (丕爻鬲賳丕丿賸丕 廿賱賶 Anaconda3)
nvidia-driver 460.73
cuda-version 11.2
pytorch 1.8.1
2. 毓賲賱賷丞 丕賱鬲孬亘賷鬲
1. 鬲孬亘賷鬲 亘乇賳丕賲噩 鬲卮睾賷賱 Nvidia
sudo apt install nvidia-driver-460
兀毓丿 鬲卮睾賷賱 丕賱賳馗丕賲 亘毓丿 丕賱鬲孬亘賷鬲 貙 兀丿禺賱 賮賷 Shellnvidia-smi 廿匕丕 馗賴乇鬲 丕賱氐賮丨丞 丕賱鬲丕賱賷丞 貙 賮廿賳 丕賱鬲孬亘賷鬲 賳丕噩丨 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 460.73.01 Driver Version: 460.73.01 CUDA Version: 11.2 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 GeForce MX150 Off | 00000000:01:00.0 Off | N/A | | N/A 52C P0 N/A / N/A | 278MiB / 2002MiB | 7% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=============================================================================| | 0 N/A N/A 927 G /usr/lib/xorg/Xorg 45MiB | | 0 N/A N/A 1694 G /usr/lib/xorg/Xorg 109MiB | | 0 N/A N/A 1878 G /usr/bin/gnome-shell 37MiB | | 0 N/A N/A 2549 G .../files/bin/baidu-qimpanel 15MiB | | 0 N/A N/A 3359 G ...AAAAAAAAA= --shared-files 62MiB | +-----------------------------------------------------------------------------+ 2. 鬲孬亘賷鬲 亘賷卅丞 兀賳丕賰賵賳丿丕毓賳賵丕賳 丕賱賲賵賯毓 丕賱乇爻賲賷:https://www.anaconda.com/products/individual#Downloads 丨丿丿 丕賱廿氐丿丕乇 丕賱賲賯丕亘賱 賱賱鬲賳夭賷賱 丕賱鬲賳夭賷賱 賴賵 賲賱賮 sh 貙 賲孬賱Anaconda3-2020.11-Linux-x86_64.sh 賳馗乇賸丕 賱毓丿賲 賵噩賵丿 廿匕賳 貙 賳丨鬲丕噩 廿賱賶 賲賳丨賴 兀匕賵賳丕鬲 丕賱鬲賳賮賷匕 兀賵賱丕賸 貙 孬賲 丕賱鬲孬亘賷鬲 sudo chmod 774 Anaconda3-2020.11-Linux-x86_64.sh bash Anaconda3-2020.11-Linux-x86_64.sh賱毓賲賱賷丞 丕賱鬲孬亘賷鬲 丕賱鬲賮氐賷賱賷丞 貙 賷乇噩賶 丕賱乇噩賵毓 廿賱賶鬲孬亘賷鬲 Ubuntu 賲賳 Anaconda3 3. 鬲孬亘賷鬲 CUDA 賵 CUDNN毓賳賵丕賳 丕賱賲賵賯毓 丕賱乇爻賲賷 賱鬲孬亘賷鬲 CUDA:https://developer.nvidia.com/cuda-toolkit-archive 丨丿丿 鬲孬亘賷鬲 丕賱廿氐丿丕乇 丕賱賲胤賱賵亘 貙 賴賳丕 丕禺鬲乇CUDA Toolkit 11.2.2 (March 2021), Versioned Online Documentation賴匕丕 丕賳賯乇 賮賷 賵丕噩賴 胤乇賯 鬲孬亘賷鬲 賲賮氐賱丞 丕禺鬲乇 丕賱賳馗丕賲 丕賱兀爻丕爻賷 丕賱禺丕氐 亘賰 兀賵賱丕賸. 賴賳丕 兀賳丕 Ubuntu20.04 孬賲 爻賷賰賵賳 賴賳丕賰 兀賲乇 鬲孬亘賷鬲 兀丿賳丕賴 貙 孬賲 賷鬲賲 鬲賳賮賷匕賴 賵賮賯賸丕 賱賱兀賲乇: wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.0-460.27.04-1_amd64.deb sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub sudo apt-get update sudo apt-get -y install cuda亘毓丿 丕賱鬲賳賮賷匕/usr/local 爻鬲馗賴乇 賲賱賮丕鬲 CUDA 賮賷 丕賱丿賱賷賱 孬賲 兀囟賮 CUDA 廿賱賶 賲鬲睾賷乇 丕賱亘賷卅丞 (廿囟丕賮丞 廿賱賶 丕賱廿囟丕賮丞/etc/profile 兀賵 丿賱賷賱 賲賳夭賱賷.bashrc 賰賱丕賴賲丕 賷賲賰賳 兀賳 賷賰賵賳): export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH export CUDA_HOME=/usr/local/cuda export PATH=${CUDA_HOME}/lib64:$PATH鬲賳賮賷匕 賮賷 卮賱source /etc/profile 兀賵 source ~/.bashrc 鬲丨丿賷孬 賲鬲睾賷乇丕鬲 丕賱亘賷卅丞 貙 丕賱賲丿禺賱丕鬲 賮賷 卮賱nvcc --version 賷賲賰賳賰 乇丐賷丞 賳噩丕丨 乇賯賲 丕賱廿氐丿丕乇 (廿匕丕 賱賲 賷賳噩丨 貙 賷賲賰賳賰 賲丨丕賵賱丞 廿毓丕丿丞 鬲卮睾賷賱 丕賱賰賲亘賷賵鬲乇) 4. 鬲孬亘賷鬲 Pytorch毓賳賵丕賳 賲丨胤丞 賲乇丌丞 Tsinghua Pytorch's Tsinghua:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 丕禺鬲乇 丕賱廿氐丿丕乇 丕賱賲乇丕丿 鬲孬亘賷鬲賴. 賷鬲賲 丕爻鬲禺丿丕賲 胤乇賷賯丞 丕賱鬲孬亘賷鬲 丕賱賲丨賱賷丞 賴賳丕: wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2 conda install --offline pytorch-1.8.1-py3.8_cuda11.1_cudnn8.0.5_0.tar.bz2丕禺鬲亘丕乇 賮賷 亘賷孬賵賳 亘毓丿 丕賱鬲孬亘賷鬲 賷賲賰賳賰 胤亘丕毓丞 乇賯賲 丕賱廿氐丿丕乇 丕賱毓丕丿賷 貙 賲賲丕 賷卮賷乇 廿賱賶 兀賳 丕賱鬲孬亘賷鬲 賳丕噩丨. |
| CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |